In your final repo, there should be an R markdown file that organizes all computational steps for evaluating your proposed Facial Expression Recognition framework.

This file is currently a template for running evaluation experiments. You should update it according to your codes but following precisely the same structure.

if(!require("EBImage")){
  install.packages("BiocManager")
  BiocManager::install("EBImage")
}
if(!require("R.matlab")){
  install.packages("R.matlab")
}
if(!require("readxl")){
  install.packages("readxl")
}

if(!require("dplyr")){
  install.packages("dplyr")
}
if(!require("readxl")){
  install.packages("readxl")
}

if(!require("ggplot2")){
  install.packages("ggplot2")
}

if(!require("caret")){
  install.packages("caret")
}

if(!require("glmnet")){
  install.packages("glmnet")
}

if(!require("WeightedROC")){
  install.packages("WeightedROC")
}

if(!require("e1071")){
  install.packages("e1071")
}

if(!require("xgboost")){
  install.packages("xgboost")
}

if(!require("randomForest")){
  install.packages("randomForest")
}
if(!require("gbm")){
  install.packages("gbm")
}

library(R.matlab)
library(readxl)
library(dplyr)
library(EBImage)
library(ggplot2)
library(caret)
library(glmnet)
library(WeightedROC)
library(e1071)
library(xgboost)
library(randomForest)
library(gbm)

Step 0 set work directories

set.seed(2020)
# setwd("~/Project3-FacialEmotionRecognition/doc")
# here replace it with your own path or manually set it in RStudio to where this rmd file is located. 
# use relative path for reproducibility

Provide directories for training images. Training images and Training fiducial points will be in different subfolders.

test_dir <- "../data/test_set_predict/" # This will be modified for different data sets.
test_image_dir <- paste(test_dir, "images/", sep="")
test_pt_dir <- paste(test_dir,  "points/", sep="")
test_label_path <- paste(test_dir, "label_prediction.csv", sep="") 

Step 1: set up controls for evaluation experiments.

In this chunk, we have a set of controls for the evaluation experiments.

run.cv.baseline <- FALSE # run cross-validation on the gbm baseline
sample.reweight <- TRUE # run sample reweighting in model training
K <- 5  # number of CV folds
run.feature.train <- TRUE # process features for training set
run.test <- TRUE # run evaluation on an independent test set
run.feature.test <- TRUE # process features for test set
run.cv.xgboost <- FALSE

Using cross-validation or independent test set evaluation, we compare the performance of models with different specifications. In this Starter Code, we tune parameter lambda (the amount of shrinkage) for logistic regression with LASSO penalty.


#gbm parameters tuning:
n.trees = c(10,50,100,200)
shrinkage = c(0.01,0.05,0.1,0.15)

#xgboost parameters tuning
n_iterations <- 100

Step 2: import data and train-test split

set.seed(2020)
#train-test split
info <- read.csv(test_label_path)
n <- nrow(info)
#train_idx <- sample(info$Index, n_train, replace = F)
test_idx <- seq(1, n, by=1)

If you choose to extract features from images, such as using Gabor filter, R memory will exhaust all images are read together. The solution is to repeat reading a smaller batch(e.g 100) and process them.

n_files <- length(list.files(test_image_dir))

image_list <- list()
for(i in 1:100){
   image_list[[i]] <- readImage(paste0(test_image_dir, sprintf("%04d", i), ".jpg"))
}

Fiducial points are stored in matlab format. In this step, we read them and store them in a list.

#function to read fiducial points
#input: index
#output: matrix of fiducial points corresponding to the index
readMat.matrix <- function(index){
     return(round(readMat(paste0(test_pt_dir, sprintf("%04d", index), ".mat"))[[1]],0))
}

#load fiducial points
fiducial_pt_list <- lapply(1:n_files, readMat.matrix)
save(fiducial_pt_list, file="../output/fiducial_pt_list_test.RData")

Step 3: construct features and responses

Figure1

feature.R should be the wrapper for all your feature engineering functions and options. The function feature( ) should have options that correspond to different scenarios for your project and produces an R object that contains features and responses that are required by all the models you are going to evaluate later.

source("../lib/feature.R")
#tm_feature_train <- NA
#if(run.feature.train){
#  tm_feature_train <- system.time(dat_train <- feature(fiducial_pt_list, train_idx))
#  save(dat_train, file="../output/feature_train.RData")
#}else{
#load(file="../output/feature_train.RData")
#}

tm_feature_test <- NA
if(run.feature.test){
  tm_feature_test <- system.time(dat_test <- feature(fiducial_pt_list, test_idx))
  save(dat_test, file="../output/feature_test.RData")
}else{
  load(file="../output/feature_test.RData")
}

Step 4: Train a classification model with training features and responses

Call the train model and test model from library.

train.R and test.R should be wrappers for all your model training steps and your classification/prediction steps.

#source("../lib/train_gbm.R")
source("../lib/test_gbm.R")
#source("../lib/fit_train_xgboost.R")

Run test on test images

*Baseline model

load("../output/res_cv_gbm.RData")
best_n.trees = as.numeric(res_cv_gbm[which.min(res_cv_gbm$mean_error),1])
best_shrinkage = as.numeric(res_cv_gbm[which.min(res_cv_gbm$mean_error),2])

source("../lib/test_gbm.R")
tm_test_baseline = NA
feature_test <- as.matrix(dat_test[, -6006])

if(run.test){
  load(file="../output/fit_train_baseline.RData")
  tm_test_baseline <- system.time({
    label_pred_baseline <- as.integer(test_gbm(fit_train_baseline,feature_test, best_n.trees, best_shrinkage, pred.type = 'link'));
    label_pred_baseline <- label_pred_baseline - 1;
  prob_pred_baseline <- test_gbm(fit_train_baseline, feature_test,best_n.trees, best_shrinkage, pred.type = 'response')})
}

*Prediction by baseline


predict_baseline <- as.data.frame(label_pred_baseline)
#write.csv(baseline, "label.csv", row.names = F)

*XGBoost (Improved model)

tm_test_xgb = NA
feature_test <- as.matrix(dat_test[, -6006])
colnames(feature_test) <- NULL
if(run.test){
  load(file="../output/fit_train_xgb.RData")
  tm_test_xgb <- system.time({label_pred_xgb <- predict(fit_train_xgb, feature_test, pred.type = 'class');
                          label_pred_xgb[label_pred_xgb >= 0.5] <- 1;
                          label_pred_xgb[label_pred_xgb < 0.5] <-  2;
                          label_pred_xgb  <- label_pred_xgb - 1
                          prob_pred_xgb <- predict(fit_train_xgb, feature_test, pred.type = 'response')})
}

*Prediction by Improved model

predict_xgb <- as.data.frame(label_pred_xgb)

*Combining the datasets into one and writing the .csv

index <- as.data.frame(1:2000)
output_csv <- cbind(index, predict_baseline, predict_xgb)
names(output_csv) <- c("index", "baseline model", "xgboost (improved) model")
write.csv(output_csv, "test_prediction.csv", row.names = F)

Evaluation

*Baseline Model

## reweight the test data to represent a balanced label distribution
label_test <- as.integer(dat_test$label)
weight_test <- rep(NA, length(label_test))
for (v in unique(label_test)){
  weight_test[label_test == v] = 0.5 * length(label_test) / length(label_test[label_test == v])
}

accu_baseline <- mean(label_pred_baseline == label_test)
tpr.fpr.baseline <- WeightedROC(prob_pred_baseline, label_test, weight_test)
auc_baseline <- WeightedAUC(tpr.fpr.baseline)


cat("The accuracy of model GBM: with n.trees=",best_n.trees,"and shrinkage =", best_shrinkage, "is", accu_baseline*100, "%.\n")
cat("The AUC of model GBM: with n.trees=", best_n.trees,"and shrinkage =", best_shrinkage, "is", auc_baseline, ".\n")

*Improved Model

label_test <- as.integer(dat_test$label)
label_test_xgb <- label_test
label_test_xgb[label_test_xgb==2] = 0

weight_test <- rep(NA, length(label_test_xgb))
for (v in unique(label_test_xgb)){
  weight_test[label_test_xgb == v] = 0.5 * length(label_test_xgb) / length(label_test_xgb[label_test_xgb == v])
}

accu_xgb <- mean((label_pred_xgb == label_test_xgb))
tpr.fpr_xgb <- WeightedROC(prob_pred_xgb, label_test_xgb, weight_test)
auc_xgb <- WeightedAUC(tpr.fpr_xgb)


cat("The accuracy of the XGBoost model:", "is", accu_xgb*100, "%.\n")
cat("The AUC of the XGBoost model:", "is", auc_xgb, ".\n")

Summarize Running Time

Prediction performance matters, so does the running times for constructing features and for training the model, especially when the computation resource is limited.

*Baseline Model

cat("Time for constructing training features=", tm_feature_train[1], "s \n")
cat("Time for constructing testing features=", tm_feature_test[1], "s \n")
cat("Time for training model=", tm_train_baseline[1], "s \n") 
cat("Time for testing model=", tm_test_baseline[1], "s \n")
cat("Time for constructing training features=", tm_feature_train[1], "s \n")
cat("Time for constructing testing features=", tm_feature_test[1], "s \n")
cat("Time for training model=", xgb_train_time[1], "s \n") 
cat("Time for testing model=", tm_test_xgb[1], "s \n")

###Reference - Du, S., Tao, Y., & Martinez, A. M. (2014). Compound facial expressions of emotion. Proceedings of the National Academy of Sciences, 111(15), E1454-E1462.

LS0tCnRpdGxlOiAiVGVzdCIKYXV0aG9yOiAiRGFpenkgTGFtLCBQZXRlciBLd2F1aywgUWl6aGVuIFlhbmcsIEVsbGVuIENoZW4sIERhcnlsIEtvdyIKb3V0cHV0OgogIHBkZl9kb2N1bWVudDogZGVmYXVsdAogIGh0bWxfbm90ZWJvb2s6IGRlZmF1bHQKLS0tCgpJbiB5b3VyIGZpbmFsIHJlcG8sIHRoZXJlIHNob3VsZCBiZSBhbiBSIG1hcmtkb3duIGZpbGUgdGhhdCBvcmdhbml6ZXMgKiphbGwgY29tcHV0YXRpb25hbCBzdGVwcyoqIGZvciBldmFsdWF0aW5nIHlvdXIgcHJvcG9zZWQgRmFjaWFsIEV4cHJlc3Npb24gUmVjb2duaXRpb24gZnJhbWV3b3JrLiAKClRoaXMgZmlsZSBpcyBjdXJyZW50bHkgYSB0ZW1wbGF0ZSBmb3IgcnVubmluZyBldmFsdWF0aW9uIGV4cGVyaW1lbnRzLiBZb3Ugc2hvdWxkIHVwZGF0ZSBpdCBhY2NvcmRpbmcgdG8geW91ciBjb2RlcyBidXQgZm9sbG93aW5nIHByZWNpc2VseSB0aGUgc2FtZSBzdHJ1Y3R1cmUuIAoKYGBge3IgbWVzc2FnZT1GQUxTRX0KaWYoIXJlcXVpcmUoIkVCSW1hZ2UiKSl7CiAgaW5zdGFsbC5wYWNrYWdlcygiQmlvY01hbmFnZXIiKQogIEJpb2NNYW5hZ2VyOjppbnN0YWxsKCJFQkltYWdlIikKfQppZighcmVxdWlyZSgiUi5tYXRsYWIiKSl7CiAgaW5zdGFsbC5wYWNrYWdlcygiUi5tYXRsYWIiKQp9CmlmKCFyZXF1aXJlKCJyZWFkeGwiKSl7CiAgaW5zdGFsbC5wYWNrYWdlcygicmVhZHhsIikKfQoKaWYoIXJlcXVpcmUoImRwbHlyIikpewogIGluc3RhbGwucGFja2FnZXMoImRwbHlyIikKfQppZighcmVxdWlyZSgicmVhZHhsIikpewogIGluc3RhbGwucGFja2FnZXMoInJlYWR4bCIpCn0KCmlmKCFyZXF1aXJlKCJnZ3Bsb3QyIikpewogIGluc3RhbGwucGFja2FnZXMoImdncGxvdDIiKQp9CgppZighcmVxdWlyZSgiY2FyZXQiKSl7CiAgaW5zdGFsbC5wYWNrYWdlcygiY2FyZXQiKQp9CgppZighcmVxdWlyZSgiZ2xtbmV0IikpewogIGluc3RhbGwucGFja2FnZXMoImdsbW5ldCIpCn0KCmlmKCFyZXF1aXJlKCJXZWlnaHRlZFJPQyIpKXsKICBpbnN0YWxsLnBhY2thZ2VzKCJXZWlnaHRlZFJPQyIpCn0KCmlmKCFyZXF1aXJlKCJlMTA3MSIpKXsKICBpbnN0YWxsLnBhY2thZ2VzKCJlMTA3MSIpCn0KCmlmKCFyZXF1aXJlKCJ4Z2Jvb3N0IikpewogIGluc3RhbGwucGFja2FnZXMoInhnYm9vc3QiKQp9CgppZighcmVxdWlyZSgicmFuZG9tRm9yZXN0IikpewogIGluc3RhbGwucGFja2FnZXMoInJhbmRvbUZvcmVzdCIpCn0KaWYoIXJlcXVpcmUoImdibSIpKXsKICBpbnN0YWxsLnBhY2thZ2VzKCJnYm0iKQp9CgpsaWJyYXJ5KFIubWF0bGFiKQpsaWJyYXJ5KHJlYWR4bCkKbGlicmFyeShkcGx5cikKbGlicmFyeShFQkltYWdlKQpsaWJyYXJ5KGdncGxvdDIpCmxpYnJhcnkoY2FyZXQpCmxpYnJhcnkoZ2xtbmV0KQpsaWJyYXJ5KFdlaWdodGVkUk9DKQpsaWJyYXJ5KGUxMDcxKQpsaWJyYXJ5KHhnYm9vc3QpCmxpYnJhcnkocmFuZG9tRm9yZXN0KQpsaWJyYXJ5KGdibSkKYGBgCgojIyMgU3RlcCAwIHNldCB3b3JrIGRpcmVjdG9yaWVzCmBgYHtyIHdrZGlyLCBldmFsPUZBTFNFfQpzZXQuc2VlZCgyMDIwKQojIHNldHdkKCJ+L1Byb2plY3QzLUZhY2lhbEVtb3Rpb25SZWNvZ25pdGlvbi9kb2MiKQojIGhlcmUgcmVwbGFjZSBpdCB3aXRoIHlvdXIgb3duIHBhdGggb3IgbWFudWFsbHkgc2V0IGl0IGluIFJTdHVkaW8gdG8gd2hlcmUgdGhpcyBybWQgZmlsZSBpcyBsb2NhdGVkLiAKIyB1c2UgcmVsYXRpdmUgcGF0aCBmb3IgcmVwcm9kdWNpYmlsaXR5CmBgYAoKUHJvdmlkZSBkaXJlY3RvcmllcyBmb3IgdHJhaW5pbmcgaW1hZ2VzLiBUcmFpbmluZyBpbWFnZXMgYW5kIFRyYWluaW5nIGZpZHVjaWFsIHBvaW50cyB3aWxsIGJlIGluIGRpZmZlcmVudCBzdWJmb2xkZXJzLiAKYGBge3J9CnRlc3RfZGlyIDwtICIuLi9kYXRhL3Rlc3Rfc2V0X3ByZWRpY3QvIiAjIFRoaXMgd2lsbCBiZSBtb2RpZmllZCBmb3IgZGlmZmVyZW50IGRhdGEgc2V0cy4KdGVzdF9pbWFnZV9kaXIgPC0gcGFzdGUodGVzdF9kaXIsICJpbWFnZXMvIiwgc2VwPSIiKQp0ZXN0X3B0X2RpciA8LSBwYXN0ZSh0ZXN0X2RpciwgICJwb2ludHMvIiwgc2VwPSIiKQp0ZXN0X2xhYmVsX3BhdGggPC0gcGFzdGUodGVzdF9kaXIsICJsYWJlbF9wcmVkaWN0aW9uLmNzdiIsIHNlcD0iIikgCmBgYAoKIyMjIFN0ZXAgMTogc2V0IHVwIGNvbnRyb2xzIGZvciBldmFsdWF0aW9uIGV4cGVyaW1lbnRzLgoKSW4gdGhpcyBjaHVuaywgd2UgaGF2ZSBhIHNldCBvZiBjb250cm9scyBmb3IgdGhlIGV2YWx1YXRpb24gZXhwZXJpbWVudHMuIAoKKyAoVC9GKSBjcm9zcy12YWxpZGF0aW9uIG9uIHRoZSB0cmFpbmluZyBzZXQKKyAoVC9GKSByZXdlaWdodGluZyB0aGUgc2FtcGxlcyBmb3IgdHJhaW5pbmcgc2V0IAorIChudW1iZXIpIEssIHRoZSBudW1iZXIgb2YgQ1YgZm9sZHMKKyAoVC9GKSBwcm9jZXNzIGZlYXR1cmVzIGZvciB0cmFpbmluZyBzZXQKKyAoVC9GKSBydW4gZXZhbHVhdGlvbiBvbiBhbiBpbmRlcGVuZGVudCB0ZXN0IHNldAorIChUL0YpIHByb2Nlc3MgZmVhdHVyZXMgZm9yIHRlc3Qgc2V0CgpgYGB7ciBleHBfc2V0dXB9CnJ1bi5jdi5iYXNlbGluZSA8LSBGQUxTRSAjIHJ1biBjcm9zcy12YWxpZGF0aW9uIG9uIHRoZSBnYm0gYmFzZWxpbmUKc2FtcGxlLnJld2VpZ2h0IDwtIFRSVUUgIyBydW4gc2FtcGxlIHJld2VpZ2h0aW5nIGluIG1vZGVsIHRyYWluaW5nCksgPC0gNSAgIyBudW1iZXIgb2YgQ1YgZm9sZHMKcnVuLmZlYXR1cmUudHJhaW4gPC0gVFJVRSAjIHByb2Nlc3MgZmVhdHVyZXMgZm9yIHRyYWluaW5nIHNldApydW4udGVzdCA8LSBUUlVFICMgcnVuIGV2YWx1YXRpb24gb24gYW4gaW5kZXBlbmRlbnQgdGVzdCBzZXQKcnVuLmZlYXR1cmUudGVzdCA8LSBUUlVFICMgcHJvY2VzcyBmZWF0dXJlcyBmb3IgdGVzdCBzZXQKcnVuLmN2LnhnYm9vc3QgPC0gRkFMU0UKYGBgCgpVc2luZyBjcm9zcy12YWxpZGF0aW9uIG9yIGluZGVwZW5kZW50IHRlc3Qgc2V0IGV2YWx1YXRpb24sIHdlIGNvbXBhcmUgdGhlIHBlcmZvcm1hbmNlIG9mIG1vZGVscyB3aXRoIGRpZmZlcmVudCBzcGVjaWZpY2F0aW9ucy4gSW4gdGhpcyBTdGFydGVyIENvZGUsIHdlIHR1bmUgcGFyYW1ldGVyIGxhbWJkYSAodGhlIGFtb3VudCBvZiBzaHJpbmthZ2UpIGZvciBsb2dpc3RpYyByZWdyZXNzaW9uIHdpdGggTEFTU08gcGVuYWx0eS4KCmBgYHtyIG1vZGVsX3NldHVwfQoKI2dibSBwYXJhbWV0ZXJzIHR1bmluZzoKbi50cmVlcyA9IGMoMTAsNTAsMTAwLDIwMCkKc2hyaW5rYWdlID0gYygwLjAxLDAuMDUsMC4xLDAuMTUpCgojeGdib29zdCBwYXJhbWV0ZXJzIHR1bmluZwpuX2l0ZXJhdGlvbnMgPC0gMTAwCgoKYGBgCgojIyMgU3RlcCAyOiBpbXBvcnQgZGF0YSBhbmQgdHJhaW4tdGVzdCBzcGxpdCAKYGBge3J9CnNldC5zZWVkKDIwMjApCiN0cmFpbi10ZXN0IHNwbGl0CmluZm8gPC0gcmVhZC5jc3YodGVzdF9sYWJlbF9wYXRoKQpuIDwtIG5yb3coaW5mbykKI3RyYWluX2lkeCA8LSBzYW1wbGUoaW5mbyRJbmRleCwgbl90cmFpbiwgcmVwbGFjZSA9IEYpCnRlc3RfaWR4IDwtIHNlcSgxLCBuLCBieT0xKQpgYGAKCklmIHlvdSBjaG9vc2UgdG8gZXh0cmFjdCBmZWF0dXJlcyBmcm9tIGltYWdlcywgc3VjaCBhcyB1c2luZyBHYWJvciBmaWx0ZXIsIFIgbWVtb3J5IHdpbGwgZXhoYXVzdCBhbGwgaW1hZ2VzIGFyZSByZWFkIHRvZ2V0aGVyLiBUaGUgc29sdXRpb24gaXMgdG8gcmVwZWF0IHJlYWRpbmcgYSBzbWFsbGVyIGJhdGNoKGUuZyAxMDApIGFuZCBwcm9jZXNzIHRoZW0uIApgYGB7cn0Kbl9maWxlcyA8LSBsZW5ndGgobGlzdC5maWxlcyh0ZXN0X2ltYWdlX2RpcikpCgppbWFnZV9saXN0IDwtIGxpc3QoKQpmb3IoaSBpbiAxOjEwMCl7CiAgIGltYWdlX2xpc3RbW2ldXSA8LSByZWFkSW1hZ2UocGFzdGUwKHRlc3RfaW1hZ2VfZGlyLCBzcHJpbnRmKCIlMDRkIiwgaSksICIuanBnIikpCn0KYGBgCgpGaWR1Y2lhbCBwb2ludHMgYXJlIHN0b3JlZCBpbiBtYXRsYWIgZm9ybWF0LiBJbiB0aGlzIHN0ZXAsIHdlIHJlYWQgdGhlbSBhbmQgc3RvcmUgdGhlbSBpbiBhIGxpc3QuCmBgYHtyIHJlYWQgZmlkdWNpYWwgcG9pbnRzfQojZnVuY3Rpb24gdG8gcmVhZCBmaWR1Y2lhbCBwb2ludHMKI2lucHV0OiBpbmRleAojb3V0cHV0OiBtYXRyaXggb2YgZmlkdWNpYWwgcG9pbnRzIGNvcnJlc3BvbmRpbmcgdG8gdGhlIGluZGV4CnJlYWRNYXQubWF0cml4IDwtIGZ1bmN0aW9uKGluZGV4KXsKICAgICByZXR1cm4ocm91bmQocmVhZE1hdChwYXN0ZTAodGVzdF9wdF9kaXIsIHNwcmludGYoIiUwNGQiLCBpbmRleCksICIubWF0IikpW1sxXV0sMCkpCn0KCiNsb2FkIGZpZHVjaWFsIHBvaW50cwpmaWR1Y2lhbF9wdF9saXN0IDwtIGxhcHBseSgxOm5fZmlsZXMsIHJlYWRNYXQubWF0cml4KQpzYXZlKGZpZHVjaWFsX3B0X2xpc3QsIGZpbGU9Ii4uL291dHB1dC9maWR1Y2lhbF9wdF9saXN0X3Rlc3QuUkRhdGEiKQpgYGAKCiMjIyBTdGVwIDM6IGNvbnN0cnVjdCBmZWF0dXJlcyBhbmQgcmVzcG9uc2VzCgorIFRoZSBmb2xsb3cgcGxvdHMgc2hvdyBob3cgcGFpcndpc2UgZGlzdGFuY2UgYmV0d2VlbiBmaWR1Y2lhbCBwb2ludHMgY2FuIHdvcmsgYXMgZmVhdHVyZSBmb3IgZmFjaWFsIGVtb3Rpb24gcmVjb2duaXRpb24uCgogICsgSW4gdGhlIGZpcnN0IGNvbHVtbiwgNzggZmlkdWNpYWxzIHBvaW50cyBvZiBlYWNoIGVtb3Rpb24gYXJlIG1hcmtlZCBpbiBvcmRlci4gCiAgKyBJbiB0aGUgc2Vjb25kIGNvbHVtbiBkaXN0cmlidXRpb25zIG9mIHZlcnRpY2FsIGRpc3RhbmNlIGJldHdlZW4gcmlnaHQgcHVwaWwoMSkgYW5kICByaWdodCBicm93IHBlYWsoMjEpIGFyZSBzaG93biBpbiAgaGlzdG9ncmFtcy4gRm9yIGV4YW1wbGUsIHRoZSBkaXN0YW5jZSBvZiBhbiBhbmdyeSBmYWNlIHRlbmRzIHRvIGJlIHNob3J0ZXIgdGhhbiB0aGF0IG9mIGEgc3VycHJpc2VkIGZhY2UuCiAgKyBUaGUgdGhpcmQgY29sdW1uIGlzIHRoZSBkaXN0cmlidXRpb25zIG9mIHZlcnRpY2FsIGRpc3RhbmNlcyBiZXR3ZWVuIHJpZ2h0IG1vdXRoIGNvcm5lcig1MCkKYW5kIHRoZSBtaWRwb2ludCBvZiB0aGUgdXBwZXIgbGlwKDUyKS4gIEZvciBleGFtcGxlLCB0aGUgZGlzdGFuY2Ugb2YgYW4gaGFwcHkgZmFjZSB0ZW5kcyB0byBiZSBzaG9ydGVyIHRoYW4gdGhhdCBvZiBhIHNhZCBmYWNlLgoKIVtGaWd1cmUxXSguLi9maWdzL2ZlYXR1cmVfdmlzdWFsaXphdGlvbi5qcGcpCgpgZmVhdHVyZS5SYCBzaG91bGQgYmUgdGhlIHdyYXBwZXIgZm9yIGFsbCB5b3VyIGZlYXR1cmUgZW5naW5lZXJpbmcgZnVuY3Rpb25zIGFuZCBvcHRpb25zLiBUaGUgZnVuY3Rpb24gYGZlYXR1cmUoIClgIHNob3VsZCBoYXZlIG9wdGlvbnMgdGhhdCBjb3JyZXNwb25kIHRvIGRpZmZlcmVudCBzY2VuYXJpb3MgZm9yIHlvdXIgcHJvamVjdCBhbmQgcHJvZHVjZXMgYW4gUiBvYmplY3QgdGhhdCBjb250YWlucyBmZWF0dXJlcyBhbmQgcmVzcG9uc2VzIHRoYXQgYXJlIHJlcXVpcmVkIGJ5IGFsbCB0aGUgbW9kZWxzIHlvdSBhcmUgZ29pbmcgdG8gZXZhbHVhdGUgbGF0ZXIuIAogIAogICsgYGZlYXR1cmUuUmAKICArIElucHV0OiBsaXN0IG9mIGltYWdlcyBvciBmaWR1Y2lhbCBwb2ludAogICsgT3V0cHV0OiBhbiBSRGF0YSBmaWxlIHRoYXQgY29udGFpbnMgZXh0cmFjdGVkIGZlYXR1cmVzIGFuZCBjb3JyZXNwb25kaW5nIHJlc3BvbnNlcwoKYGBge3IgZmVhdHVyZX0Kc291cmNlKCIuLi9saWIvZmVhdHVyZS5SIikKI3RtX2ZlYXR1cmVfdHJhaW4gPC0gTkEKI2lmKHJ1bi5mZWF0dXJlLnRyYWluKXsKIyAgdG1fZmVhdHVyZV90cmFpbiA8LSBzeXN0ZW0udGltZShkYXRfdHJhaW4gPC0gZmVhdHVyZShmaWR1Y2lhbF9wdF9saXN0LCB0cmFpbl9pZHgpKQojICBzYXZlKGRhdF90cmFpbiwgZmlsZT0iLi4vb3V0cHV0L2ZlYXR1cmVfdHJhaW4uUkRhdGEiKQojfWVsc2V7CiNsb2FkKGZpbGU9Ii4uL291dHB1dC9mZWF0dXJlX3RyYWluLlJEYXRhIikKI30KCnRtX2ZlYXR1cmVfdGVzdCA8LSBOQQppZihydW4uZmVhdHVyZS50ZXN0KXsKICB0bV9mZWF0dXJlX3Rlc3QgPC0gc3lzdGVtLnRpbWUoZGF0X3Rlc3QgPC0gZmVhdHVyZShmaWR1Y2lhbF9wdF9saXN0LCB0ZXN0X2lkeCkpCiAgc2F2ZShkYXRfdGVzdCwgZmlsZT0iLi4vb3V0cHV0L2ZlYXR1cmVfdGVzdC5SRGF0YSIpCn1lbHNlewogIGxvYWQoZmlsZT0iLi4vb3V0cHV0L2ZlYXR1cmVfdGVzdC5SRGF0YSIpCn0KCgpgYGAKCiMjIyBTdGVwIDQ6IFRyYWluIGEgY2xhc3NpZmljYXRpb24gbW9kZWwgd2l0aCB0cmFpbmluZyBmZWF0dXJlcyBhbmQgcmVzcG9uc2VzCkNhbGwgdGhlIHRyYWluIG1vZGVsIGFuZCB0ZXN0IG1vZGVsIGZyb20gbGlicmFyeS4gCgpgdHJhaW4uUmAgYW5kIGB0ZXN0LlJgIHNob3VsZCBiZSB3cmFwcGVycyBmb3IgYWxsIHlvdXIgbW9kZWwgdHJhaW5pbmcgc3RlcHMgYW5kIHlvdXIgY2xhc3NpZmljYXRpb24vcHJlZGljdGlvbiBzdGVwcy4gCgorIGB0cmFpbi5SYAogICsgSW5wdXQ6IGEgZGF0YSBmcmFtZSBjb250YWluaW5nIGZlYXR1cmVzIGFuZCBsYWJlbHMgYW5kIGEgcGFyYW1ldGVyIGxpc3QuCiAgKyBPdXRwdXQ6YSB0cmFpbmVkIG1vZGVsCisgYHRlc3QuUmAKICArIElucHV0OiB0aGUgZml0dGVkIGNsYXNzaWZpY2F0aW9uIG1vZGVsIHVzaW5nIHRyYWluaW5nIGRhdGEgYW5kIHByb2Nlc3NlZCBmZWF0dXJlcyBmcm9tIHRlc3RpbmcgaW1hZ2VzIAogICsgSW5wdXQ6IGFuIFIgb2JqZWN0IHRoYXQgY29udGFpbnMgYSB0cmFpbmVkIGNsYXNzaWZpZXIuCiAgKyBPdXRwdXQ6IHRyYWluaW5nIG1vZGVsIHNwZWNpZmljYXRpb24KCisgSW4gdGhpcyBTdGFydGVyIENvZGUsIHdlIHVzZSBsb2dpc3RpYyByZWdyZXNzaW9uIHdpdGggTEFTU08gcGVuYWx0eSB0byBkbyBjbGFzc2lmaWNhdGlvbi4gCgpgYGB7ciBsb2FkbGlifQojc291cmNlKCIuLi9saWIvdHJhaW5fZ2JtLlIiKQpzb3VyY2UoIi4uL2xpYi90ZXN0X2dibS5SIikKI3NvdXJjZSgiLi4vbGliL2ZpdF90cmFpbl94Z2Jvb3N0LlIiKQoKYGBgCgoKIyMjIFJ1biB0ZXN0IG9uIHRlc3QgaW1hZ2VzCgoqQmFzZWxpbmUgbW9kZWwKYGBge3IgYmVzdF9tb2RlbH0KbG9hZCgiLi4vb3V0cHV0L3Jlc19jdl9nYm0uUkRhdGEiKQpiZXN0X24udHJlZXMgPSBhcy5udW1lcmljKHJlc19jdl9nYm1bd2hpY2gubWluKHJlc19jdl9nYm0kbWVhbl9lcnJvciksMV0pCmJlc3Rfc2hyaW5rYWdlID0gYXMubnVtZXJpYyhyZXNfY3ZfZ2JtW3doaWNoLm1pbihyZXNfY3ZfZ2JtJG1lYW5fZXJyb3IpLDJdKQpgYGAKCmBgYHtyIHRlc3QgZ2JtfQoKc291cmNlKCIuLi9saWIvdGVzdF9nYm0uUiIpCnRtX3Rlc3RfYmFzZWxpbmUgPSBOQQpmZWF0dXJlX3Rlc3QgPC0gYXMubWF0cml4KGRhdF90ZXN0WywgLTYwMDZdKQoKaWYocnVuLnRlc3QpewogIGxvYWQoZmlsZT0iLi4vb3V0cHV0L2ZpdF90cmFpbl9iYXNlbGluZS5SRGF0YSIpCiAgdG1fdGVzdF9iYXNlbGluZSA8LSBzeXN0ZW0udGltZSh7CiAgICBsYWJlbF9wcmVkX2Jhc2VsaW5lIDwtIGFzLmludGVnZXIodGVzdF9nYm0oZml0X3RyYWluX2Jhc2VsaW5lLGZlYXR1cmVfdGVzdCwgYmVzdF9uLnRyZWVzLCBiZXN0X3Nocmlua2FnZSwgcHJlZC50eXBlID0gJ2xpbmsnKSk7CiAgICBsYWJlbF9wcmVkX2Jhc2VsaW5lIDwtIGxhYmVsX3ByZWRfYmFzZWxpbmUgLSAxOwogIHByb2JfcHJlZF9iYXNlbGluZSA8LSB0ZXN0X2dibShmaXRfdHJhaW5fYmFzZWxpbmUsIGZlYXR1cmVfdGVzdCxiZXN0X24udHJlZXMsIGJlc3Rfc2hyaW5rYWdlLCBwcmVkLnR5cGUgPSAncmVzcG9uc2UnKX0pCn0KCmBgYAoKKlByZWRpY3Rpb24gYnkgYmFzZWxpbmUKYGBge3J9CgpwcmVkaWN0X2Jhc2VsaW5lIDwtIGFzLmRhdGEuZnJhbWUobGFiZWxfcHJlZF9iYXNlbGluZSkKI3dyaXRlLmNzdihiYXNlbGluZSwgImxhYmVsLmNzdiIsIHJvdy5uYW1lcyA9IEYpCmBgYAoKKlhHQm9vc3QgKEltcHJvdmVkIG1vZGVsKQpgYGB7ciB0ZXN0IHhnYm9vc3R9CnRtX3Rlc3RfeGdiID0gTkEKZmVhdHVyZV90ZXN0IDwtIGFzLm1hdHJpeChkYXRfdGVzdFssIC02MDA2XSkKY29sbmFtZXMoZmVhdHVyZV90ZXN0KSA8LSBOVUxMCmlmKHJ1bi50ZXN0KXsKICBsb2FkKGZpbGU9Ii4uL291dHB1dC9maXRfdHJhaW5feGdiLlJEYXRhIikKICB0bV90ZXN0X3hnYiA8LSBzeXN0ZW0udGltZSh7bGFiZWxfcHJlZF94Z2IgPC0gcHJlZGljdChmaXRfdHJhaW5feGdiLCBmZWF0dXJlX3Rlc3QsIHByZWQudHlwZSA9ICdjbGFzcycpOwogICAgICAgICAgICAgICAgICAgICAgICAgIGxhYmVsX3ByZWRfeGdiW2xhYmVsX3ByZWRfeGdiID49IDAuNV0gPC0gMTsKICAgICAgICAgICAgICAgICAgICAgICAgICBsYWJlbF9wcmVkX3hnYltsYWJlbF9wcmVkX3hnYiA8IDAuNV0gPC0gIDI7CiAgICAgICAgICAgICAgICAgICAgICAgICAgbGFiZWxfcHJlZF94Z2IgIDwtIGxhYmVsX3ByZWRfeGdiIC0gMQogICAgICAgICAgICAgICAgICAgICAgICAgIHByb2JfcHJlZF94Z2IgPC0gcHJlZGljdChmaXRfdHJhaW5feGdiLCBmZWF0dXJlX3Rlc3QsIHByZWQudHlwZSA9ICdyZXNwb25zZScpfSkKfQpgYGAKCipQcmVkaWN0aW9uIGJ5IEltcHJvdmVkIG1vZGVsCmBgYHtyfQpwcmVkaWN0X3hnYiA8LSBhcy5kYXRhLmZyYW1lKGxhYmVsX3ByZWRfeGdiKQpgYGAKCipDb21iaW5pbmcgdGhlIGRhdGFzZXRzIGludG8gb25lIGFuZCB3cml0aW5nIHRoZSAuY3N2CmBgYHtyfQppbmRleCA8LSBhcy5kYXRhLmZyYW1lKDE6MjAwMCkKb3V0cHV0X2NzdiA8LSBjYmluZChpbmRleCwgcHJlZGljdF9iYXNlbGluZSwgcHJlZGljdF94Z2IpCm5hbWVzKG91dHB1dF9jc3YpIDwtIGMoImluZGV4IiwgImJhc2VsaW5lIG1vZGVsIiwgInhnYm9vc3QgKGltcHJvdmVkKSBtb2RlbCIpCndyaXRlLmNzdihvdXRwdXRfY3N2LCAidGVzdF9wcmVkaWN0aW9uLmNzdiIsIHJvdy5uYW1lcyA9IEYpCmBgYAoKCiMjIyBFdmFsdWF0aW9uCipCYXNlbGluZSBNb2RlbApgYGB7ciBiYXNlbGluZX0KIyMgcmV3ZWlnaHQgdGhlIHRlc3QgZGF0YSB0byByZXByZXNlbnQgYSBiYWxhbmNlZCBsYWJlbCBkaXN0cmlidXRpb24KbGFiZWxfdGVzdCA8LSBhcy5pbnRlZ2VyKGRhdF90ZXN0JGxhYmVsKQp3ZWlnaHRfdGVzdCA8LSByZXAoTkEsIGxlbmd0aChsYWJlbF90ZXN0KSkKZm9yICh2IGluIHVuaXF1ZShsYWJlbF90ZXN0KSl7CiAgd2VpZ2h0X3Rlc3RbbGFiZWxfdGVzdCA9PSB2XSA9IDAuNSAqIGxlbmd0aChsYWJlbF90ZXN0KSAvIGxlbmd0aChsYWJlbF90ZXN0W2xhYmVsX3Rlc3QgPT0gdl0pCn0KCmFjY3VfYmFzZWxpbmUgPC0gbWVhbihsYWJlbF9wcmVkX2Jhc2VsaW5lID09IGxhYmVsX3Rlc3QpCnRwci5mcHIuYmFzZWxpbmUgPC0gV2VpZ2h0ZWRST0MocHJvYl9wcmVkX2Jhc2VsaW5lLCBsYWJlbF90ZXN0LCB3ZWlnaHRfdGVzdCkKYXVjX2Jhc2VsaW5lIDwtIFdlaWdodGVkQVVDKHRwci5mcHIuYmFzZWxpbmUpCgoKY2F0KCJUaGUgYWNjdXJhY3kgb2YgbW9kZWwgR0JNOiB3aXRoIG4udHJlZXM9IixiZXN0X24udHJlZXMsImFuZCBzaHJpbmthZ2UgPSIsIGJlc3Rfc2hyaW5rYWdlLCAiaXMiLCBhY2N1X2Jhc2VsaW5lKjEwMCwgIiUuXG4iKQpjYXQoIlRoZSBBVUMgb2YgbW9kZWwgR0JNOiB3aXRoIG4udHJlZXM9IiwgYmVzdF9uLnRyZWVzLCJhbmQgc2hyaW5rYWdlID0iLCBiZXN0X3Nocmlua2FnZSwgImlzIiwgYXVjX2Jhc2VsaW5lLCAiLlxuIikKCgpgYGAKCipJbXByb3ZlZCBNb2RlbAoKYGBge3IgZXZhbHVhdGlvbl9YR0J9CmxhYmVsX3Rlc3QgPC0gYXMuaW50ZWdlcihkYXRfdGVzdCRsYWJlbCkKbGFiZWxfdGVzdF94Z2IgPC0gbGFiZWxfdGVzdApsYWJlbF90ZXN0X3hnYltsYWJlbF90ZXN0X3hnYj09Ml0gPSAwCgp3ZWlnaHRfdGVzdCA8LSByZXAoTkEsIGxlbmd0aChsYWJlbF90ZXN0X3hnYikpCmZvciAodiBpbiB1bmlxdWUobGFiZWxfdGVzdF94Z2IpKXsKICB3ZWlnaHRfdGVzdFtsYWJlbF90ZXN0X3hnYiA9PSB2XSA9IDAuNSAqIGxlbmd0aChsYWJlbF90ZXN0X3hnYikgLyBsZW5ndGgobGFiZWxfdGVzdF94Z2JbbGFiZWxfdGVzdF94Z2IgPT0gdl0pCn0KCmFjY3VfeGdiIDwtIG1lYW4oKGxhYmVsX3ByZWRfeGdiID09IGxhYmVsX3Rlc3RfeGdiKSkKdHByLmZwcl94Z2IgPC0gV2VpZ2h0ZWRST0MocHJvYl9wcmVkX3hnYiwgbGFiZWxfdGVzdF94Z2IsIHdlaWdodF90ZXN0KQphdWNfeGdiIDwtIFdlaWdodGVkQVVDKHRwci5mcHJfeGdiKQoKCmNhdCgiVGhlIGFjY3VyYWN5IG9mIHRoZSBYR0Jvb3N0IG1vZGVsOiIsICJpcyIsIGFjY3VfeGdiKjEwMCwgIiUuXG4iKQpjYXQoIlRoZSBBVUMgb2YgdGhlIFhHQm9vc3QgbW9kZWw6IiwgImlzIiwgYXVjX3hnYiwgIi5cbiIpCgpgYGAKCgojIyMgU3VtbWFyaXplIFJ1bm5pbmcgVGltZQpQcmVkaWN0aW9uIHBlcmZvcm1hbmNlIG1hdHRlcnMsIHNvIGRvZXMgdGhlIHJ1bm5pbmcgdGltZXMgZm9yIGNvbnN0cnVjdGluZyBmZWF0dXJlcyBhbmQgZm9yIHRyYWluaW5nIHRoZSBtb2RlbCwgZXNwZWNpYWxseSB3aGVuIHRoZSBjb21wdXRhdGlvbiByZXNvdXJjZSBpcyBsaW1pdGVkLiAKCipCYXNlbGluZSBNb2RlbApgYGB7ciBydW5uaW5nX3RpbWVfYmFzZWxpbmV9CmNhdCgiVGltZSBmb3IgY29uc3RydWN0aW5nIHRyYWluaW5nIGZlYXR1cmVzPSIsIHRtX2ZlYXR1cmVfdHJhaW5bMV0sICJzIFxuIikKY2F0KCJUaW1lIGZvciBjb25zdHJ1Y3RpbmcgdGVzdGluZyBmZWF0dXJlcz0iLCB0bV9mZWF0dXJlX3Rlc3RbMV0sICJzIFxuIikKY2F0KCJUaW1lIGZvciB0cmFpbmluZyBtb2RlbD0iLCB0bV90cmFpbl9iYXNlbGluZVsxXSwgInMgXG4iKSAKY2F0KCJUaW1lIGZvciB0ZXN0aW5nIG1vZGVsPSIsIHRtX3Rlc3RfYmFzZWxpbmVbMV0sICJzIFxuIikKYGBgCgoKKiBJbXByb3ZlZCBNb2RlbApgYGB7ciBydW5uaW5nX3RpbWVfWEdCfQpjYXQoIlRpbWUgZm9yIGNvbnN0cnVjdGluZyB0cmFpbmluZyBmZWF0dXJlcz0iLCB0bV9mZWF0dXJlX3RyYWluWzFdLCAicyBcbiIpCmNhdCgiVGltZSBmb3IgY29uc3RydWN0aW5nIHRlc3RpbmcgZmVhdHVyZXM9IiwgdG1fZmVhdHVyZV90ZXN0WzFdLCAicyBcbiIpCmNhdCgiVGltZSBmb3IgdHJhaW5pbmcgbW9kZWw9IiwgeGdiX3RyYWluX3RpbWVbMV0sICJzIFxuIikgCmNhdCgiVGltZSBmb3IgdGVzdGluZyBtb2RlbD0iLCB0bV90ZXN0X3hnYlsxXSwgInMgXG4iKQoKYGBgCgoKIyMjUmVmZXJlbmNlCi0gRHUsIFMuLCBUYW8sIFkuLCAmIE1hcnRpbmV6LCBBLiBNLiAoMjAxNCkuIENvbXBvdW5kIGZhY2lhbCBleHByZXNzaW9ucyBvZiBlbW90aW9uLiBQcm9jZWVkaW5ncyBvZiB0aGUgTmF0aW9uYWwgQWNhZGVteSBvZiBTY2llbmNlcywgMTExKDE1KSwgRTE0NTQtRTE0NjIuCgoKCgoKCgoKCgoKCgo=